# class User:
#     def __init__(self, id, name):
#         self.id = id
#         self.name = name


# 导入
from sqlalchemy import Column, String, create_engine
from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.declarative import declarative_base

# 创建对象的基类
Base = declarative_base()


# 定义User对象
class User(Base):
    # 表的名字
    __tablename__ = 'user'

    # 表的结构
    id = Column(String(20), primary_key=True)
    name = Column(String(20))


# 学校....
# class School(Base):
#     __tablename__ = 'school'
#     id = Column()
#     name = Column()


# 初始化数据库连接
# 数据库类型+数据驱动名称://用户：口令@机器地址:端口号/数据库名
engine = create_engine('mysql+pymysql://root:215442748@192.168.44.180:3306/test', echo=True)

# 创建DBSession
DBSession = sessionmaker(bind=engine)

# 创建session 对象
session = DBSession()
# 创建新User对象
new_user = User(id='7', name='Bob')

# 添加到session
session.add(new_user)

# 提交即保存到数据库
session.commit()
session.close()

# 创建Session
session = DBSession()

# 创建Query查询，filter是where条件，最后调用one() 返回唯一行，如果调用all() 则返回所有行。
user = session.query(User).filter(User.id == '5').one()
print('type', type(user))
print('name', user.name)
session.close()
